#!/usr/bin/env bash
DATA_ROOT=~/data/seg_naic/round2

rm -rf cache
mkdir cache

echo "start training"
mpirun -n 4 python ./train.py \
  --data_dir='./cache' \
  --data_url=${DATA_ROOT} \
  --device='GPU' \
  2>&1  | tee cache/train.log